Node / ReticulumProjects / MeshChatX.git / files / tests / backend / test_rncp_handler_extended.py

Displaying Raw • Download

Tff7b72import T7ee787os

Tff7b72import T7ee787shutil

Tff7b72import T7ee787tempfile

Tff7b72from T7ee787unittestT7ee787.T7ee787mock Tff7b72import Te6edf3MagicMockTb4b4b4, Te6edf3patch

Tff7b72import T7ee787pytest

Tff7b72import T7ee787RNS

Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787rncp_handler Tff7b72import Te6edf3RNCPHandler

Tf0883e@pytestTff7b72.Td2a8fffixture

Tff7b72def Td2a8fftemp_dirTb4b4b4(Tb4b4b4)Tb4b4b4:

Te6edf3dir_path Tff7b72= Te6edf3tempfileTff7b72.Td2a8ffmkdtempTb4b4b4(Tb4b4b4)

Tff7b72yield Te6edf3dir_path

Te6edf3shutilTff7b72.Td2a8ffrmtreeTb4b4b4(Te6edf3dir_pathTb4b4b4)

Tf0883e@pytestTff7b72.Td2a8fffixture

Tff7b72def Td2a8ffmock_rnsTb4b4b4(Tb4b4b4)Tb4b4b4:

T8b949e# Save real Identity class to use as base for our mock class

Te6edf3real_identity_class Tff7b72= Te6edf3RNSTff7b72.Td2a8ffIdentity

Tff7b72class T56d364MockIdentityClassTb4b4b4(Te6edf3real_identity_classTb4b4b4)Tb4b4b4:

Tff7b72def Tff7b72__init__Tb4b4b4(Tff7b72selfTb4b4b4, Tff7b72*Te6edf3argsTb4b4b4, Tff7b72*Tff7b72*Te6edf3kwargsTb4b4b4)Tb4b4b4:

Tff7b72selfTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6fftest_hash_32_bytes_long_01234567Ta5d6ff"

Tff7b72selfTff7b72.Td2a8ffhexhash Tff7b72= Tff7b72selfTff7b72.Td2a8ffhashTff7b72.Td2a8ffhexTb4b4b4(Tb4b4b4)

Tff7b72with Tb4b4b4(

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.ReticulumTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_reticulumTb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.TransportTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_transportTb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.IdentityTa5d6ff"Tb4b4b4, Te6edf3MockIdentityClassTb4b4b4)Tb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.DestinationTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_destinationTb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.ResourceTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_resourceTb4b4b4,

Te6edf3patchTb4b4b4(Ta5d6ff"Ta5d6ffRNS.LinkTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_link_classTb4b4b4,

Tb4b4b4)Tb4b4b4:

Te6edf3mock_id_instance Tff7b72= Te6edf3MockIdentityClassTb4b4b4(Tb4b4b4)

Te6edf3mock_id_instanceTff7b72.Td2a8ffget_private_key Tff7b72= Te6edf3MagicMockTb4b4b4(Te6edf3return_valueTff7b72=Ta5d6ffbTa5d6ff"Ta5d6fftest_private_keyTa5d6ff"Tb4b4b4)

Tff7b72with Tb4b4b4(

Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3MockIdentityClassTb4b4b4, Ta5d6ff"Ta5d6fffrom_fileTa5d6ff"Tb4b4b4, Te6edf3return_valueTff7b72=Te6edf3mock_id_instanceTb4b4b4)Tb4b4b4,

Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3MockIdentityClassTb4b4b4, Ta5d6ff"Ta5d6ffrecallTa5d6ff"Tb4b4b4, Te6edf3return_valueTff7b72=Te6edf3mock_id_instanceTb4b4b4)Tb4b4b4,

Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(

Te6edf3MockIdentityClassTb4b4b4,

Ta5d6ff"Ta5d6fffrom_bytesTa5d6ff"Tb4b4b4,

Te6edf3return_valueTff7b72=Te6edf3mock_id_instanceTb4b4b4,

Tb4b4b4)Tb4b4b4,

Tb4b4b4)Tb4b4b4:

Te6edf3mock_dest_instance Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3mock_destinationTff7b72.Td2a8ffreturn_value Tff7b72= Te6edf3mock_dest_instance

Te6edf3mock_link_instance Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3mock_link_classTff7b72.Td2a8ffreturn_value Tff7b72= Te6edf3mock_link_instance

Te6edf3mock_link_instanceTff7b72.Td2a8ffstatus Tff7b72= Te6edf3RNSTff7b72.Td2a8ffLinkTff7b72.Td2a8ffACTIVE

Te6edf3mock_resource_instance Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3mock_resource_instanceTff7b72.Td2a8ffstatus Tff7b72= T79c0ff2 T8b949e# COMPLETE

Te6edf3mock_resource_instanceTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffres_hashTa5d6ff"

Te6edf3mock_resourceTff7b72.Td2a8ffreturn_value Tff7b72= Te6edf3mock_resource_instance

Te6edf3mock_resourceTff7b72.Td2a8ffCOMPLETE Tff7b72= T79c0ff2

Te6edf3mock_resourceTff7b72.Td2a8ffFAILED Tff7b72= T79c0ff3

Te6edf3mock_transportTff7b72.Td2a8ffactive_links Tff7b72= Tb4b4b4[Tb4b4b4]

Te6edf3mock_transportTff7b72.Td2a8ffhas_pathTff7b72.Td2a8ffreturn_value Tff7b72= Tff7b72True

Tff7b72yield Tb4b4b4{

Ta5d6ff"Ta5d6ffReticulumTa5d6ff"Tb4b4b4: Te6edf3mock_reticulumTb4b4b4,

Ta5d6ff"Ta5d6ffTransportTa5d6ff"Tb4b4b4: Te6edf3mock_transportTb4b4b4,

Ta5d6ff"Ta5d6ffIdentityTa5d6ff"Tb4b4b4: Te6edf3MockIdentityClassTb4b4b4,

Ta5d6ff"Ta5d6ffDestinationTa5d6ff"Tb4b4b4: Te6edf3mock_destinationTb4b4b4,

Ta5d6ff"Ta5d6ffResourceTa5d6ff"Tb4b4b4: Te6edf3mock_resourceTb4b4b4,

Ta5d6ff"Ta5d6ffLinkTa5d6ff"Tb4b4b4: Te6edf3mock_link_classTb4b4b4,

Ta5d6ff"Ta5d6fflink_instanceTa5d6ff"Tb4b4b4: Te6edf3mock_link_instanceTb4b4b4,

Ta5d6ff"Ta5d6ffid_instanceTa5d6ff"Tb4b4b4: Te6edf3mock_id_instanceTb4b4b4,

Ta5d6ff"Ta5d6ffdest_instanceTa5d6ff"Tb4b4b4: Te6edf3mock_dest_instanceTb4b4b4,

Tb4b4b4}

Tff7b72def Td2a8fftest_rncp_handler_initTb4b4b4(Te6edf3mock_rnsTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3handler Tff7b72= Te6edf3RNCPHandlerTb4b4b4(Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffReticulumTa5d6ff"Tb4b4b4]Tb4b4b4, Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffid_instanceTa5d6ff"Tb4b4b4]Tb4b4b4, Te6edf3temp_dirTb4b4b4)

Tff7b72assert Te6edf3handlerTff7b72.Td2a8ffreticulum Tff7b72== Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffReticulumTa5d6ff"Tb4b4b4]

Tff7b72assert Te6edf3handlerTff7b72.Td2a8ffidentity Tff7b72== Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffid_instanceTa5d6ff"Tb4b4b4]

Tff7b72assert Te6edf3handlerTff7b72.Td2a8ffstorage_dir Tff7b72== Te6edf3temp_dir

Tff7b72def Td2a8fftest_setup_receive_destinationTb4b4b4(Te6edf3mock_rnsTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3handler Tff7b72= Te6edf3RNCPHandlerTb4b4b4(Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffReticulumTa5d6ff"Tb4b4b4]Tb4b4b4, Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffid_instanceTa5d6ff"Tb4b4b4]Tb4b4b4, Te6edf3temp_dirTb4b4b4)

Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffReticulumTa5d6ff"Tb4b4b4]Tff7b72.Td2a8ffidentitypath Tff7b72= Te6edf3temp_dir

Te6edf3_ Tff7b72= Te6edf3handlerTff7b72.Td2a8ffsetup_receive_destinationTb4b4b4(

Te6edf3allowed_hashesTff7b72=Tb4b4b4[Ta5d6ff"Ta5d6ffabc123def456Ta5d6ff"Tb4b4b4]Tb4b4b4,

Te6edf3fetch_allowedTff7b72=Tff7b72TrueTb4b4b4,

Te6edf3fetch_jailTff7b72=Te6edf3temp_dirTb4b4b4,

Tb4b4b4)

Tff7b72assert Te6edf3handlerTff7b72.Td2a8ffreceive_destination Tff7b72is Tff7b72not Tff7b72None

Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffDestinationTa5d6ff"Tb4b4b4]Tff7b72.Td2a8ffassert_calledTb4b4b4(Tb4b4b4)

Tff7b72assert Te6edf3handlerTff7b72.Td2a8ffallowed_identity_hashes Tff7b72== Tb4b4b4[Tffa657bytesTff7b72.Td2a8fffromhexTb4b4b4(Ta5d6ff"Ta5d6ffabc123def456Ta5d6ff"Tb4b4b4)Tb4b4b4]

Tff7b72assert Te6edf3handlerTff7b72.Td2a8fffetch_jail Tff7b72== Te6edf3temp_dir

Tff7b72def Td2a8fftest_receive_resource_callbackTb4b4b4(Te6edf3mock_rnsTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3handler Tff7b72= Te6edf3RNCPHandlerTb4b4b4(Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffReticulumTa5d6ff"Tb4b4b4]Tb4b4b4, Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffid_instanceTa5d6ff"Tb4b4b4]Tb4b4b4, Te6edf3temp_dirTb4b4b4)

Te6edf3handlerTff7b72.Td2a8ffallowed_identity_hashes Tff7b72= Tb4b4b4[Ta5d6ffbTa5d6ff"Ta5d6ffallowed_hashTa5d6ff"Tb4b4b4]

Te6edf3mock_resource Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3mock_link Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3mock_remote_id Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3mock_remote_idTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffallowed_hashTa5d6ff"

Te6edf3mock_linkTff7b72.Td2a8ffget_remote_identityTff7b72.Td2a8ffreturn_value Tff7b72= Te6edf3mock_remote_id

Te6edf3mock_resourceTff7b72.Td2a8fflink Tff7b72= Te6edf3mock_link

T8b949e# Allowed

Tff7b72assert Te6edf3handlerTff7b72.Td2a8ff_receive_resource_callbackTb4b4b4(Te6edf3mock_resourceTb4b4b4) Tff7b72is Tff7b72True

T8b949e# Not allowed

Te6edf3mock_remote_idTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffother_hashTa5d6ff"

Tff7b72assert Te6edf3handlerTff7b72.Td2a8ff_receive_resource_callbackTb4b4b4(Te6edf3mock_resourceTb4b4b4) Tff7b72is Tff7b72False

Tff7b72def Td2a8fftest_receive_resource_concluded_successTb4b4b4(Te6edf3mock_rnsTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3handler Tff7b72= Te6edf3RNCPHandlerTb4b4b4(Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffReticulumTa5d6ff"Tb4b4b4]Tb4b4b4, Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffid_instanceTa5d6ff"Tb4b4b4]Tb4b4b4, Te6edf3temp_dirTb4b4b4)

Te6edf3mock_resource Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3mock_resourceTff7b72.Td2a8ffstatus Tff7b72= Te6edf3RNSTff7b72.Td2a8ffResourceTff7b72.Td2a8ffCOMPLETE

Te6edf3mock_resourceTff7b72.Td2a8ffhash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffresource_hashTa5d6ff"

Te6edf3mock_resourceTff7b72.Td2a8ffmetadata Tff7b72= Tb4b4b4{Ta5d6ff"Ta5d6ffnameTa5d6ff"Tb4b4b4: Ta5d6ffbTa5d6ff"Ta5d6fftest_file.txtTa5d6ff"Tb4b4b4}

T8b949e# Create dummy source file

Te6edf3source_file Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6fftemp_resource_dataTa5d6ff"Tb4b4b4)

Tff7b72with Tffa657openTb4b4b4(Te6edf3source_fileTb4b4b4, Ta5d6ff"Ta5d6ffwTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:

Te6edf3fTff7b72.Td2a8ffwriteTb4b4b4(Ta5d6ff"Ta5d6fftest dataTa5d6ff"Tb4b4b4)

Te6edf3mock_resourceTff7b72.Td2a8ffdataTff7b72.Td2a8ffname Tff7b72= Te6edf3source_file

Te6edf3handlerTff7b72.Td2a8ffactive_transfersTb4b4b4[Ta5d6ff"Ta5d6ff7265736f757263655f68617368Ta5d6ff"Tb4b4b4] Tff7b72= Tb4b4b4{Ta5d6ff"Ta5d6ffstatusTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6ffreceivingTa5d6ff"Tb4b4b4}

Te6edf3handlerTff7b72.Td2a8ff_receive_resource_concludedTb4b4b4(Te6edf3mock_resourceTb4b4b4)

T8b949e# Check if file was moved to rncp_received

Te6edf3received_dir Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffrncp_receivedTa5d6ff"Tb4b4b4)

Tff7b72assert Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffexistsTb4b4b4(Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3received_dirTb4b4b4, Ta5d6ff"Ta5d6fftest_file.txtTa5d6ff"Tb4b4b4)Tb4b4b4)

Tff7b72assert Tb4b4b4(

Te6edf3handlerTff7b72.Td2a8ffactive_transfersTb4b4b4[Ta5d6ff"Ta5d6ff7265736f757263655f68617368Ta5d6ff"Tb4b4b4]Tb4b4b4[Ta5d6ff"Ta5d6ffstatusTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffcompletedTa5d6ff"

Tb4b4b4)

Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio

Tff7b72async Tff7b72def Td2a8fftest_send_file_successTb4b4b4(Te6edf3mock_rnsTb4b4b4, Te6edf3temp_dirTb4b4b4)Tb4b4b4:

Te6edf3handler Tff7b72= Te6edf3RNCPHandlerTb4b4b4(Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffReticulumTa5d6ff"Tb4b4b4]Tb4b4b4, Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffid_instanceTa5d6ff"Tb4b4b4]Tb4b4b4, Te6edf3temp_dirTb4b4b4)

Te6edf3test_file Tff7b72= Te6edf3osTff7b72.Td2a8ffpathTff7b72.Td2a8ffjoinTb4b4b4(Te6edf3temp_dirTb4b4b4, Ta5d6ff"Ta5d6ffsend_me.txtTa5d6ff"Tb4b4b4)

Tff7b72with Tffa657openTb4b4b4(Te6edf3test_fileTb4b4b4, Ta5d6ff"Ta5d6ffwTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:

Te6edf3fTff7b72.Td2a8ffwriteTb4b4b4(Ta5d6ff"Ta5d6ffpayloadTa5d6ff"Tb4b4b4)

T8b949e# Mocking the async behavior

Te6edf3result Tff7b72= Tff7b72await Te6edf3handlerTff7b72.Td2a8ffsend_fileTb4b4b4(Ta5d6ffbTa5d6ff"Ta5d6ffdest_hashTa5d6ff"Tb4b4b4, Te6edf3test_fileTb4b4b4, Te6edf3timeoutTff7b72=T79c0ff10Tb4b4b4)

Tff7b72assert Te6edf3resultTb4b4b4[Ta5d6ff"Ta5d6ffstatusTa5d6ff"Tb4b4b4] Tff7b72== Ta5d6ff"Ta5d6ffcompletedTa5d6ff"

Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffLinkTa5d6ff"Tb4b4b4]Tff7b72.Td2a8ffassert_calledTb4b4b4(Tb4b4b4)

Te6edf3mock_rnsTb4b4b4[Ta5d6ff"Ta5d6ffResourceTa5d6ff"Tb4b4b4]Tff7b72.Td2a8ffassert_calledTb4b4b4(Tb4b4b4)